package com.arceus.leetcode;

import org.junit.Test;

/**
 * @author : iron
 * @version : 1.0.0
 * @date : 10:17 AM 2018/11/6
 */

public class T203 {


    public ListNode removeElements(ListNode head, int val) {
        if (head.next == null) {
            if (head.val == val) {
                return null;
            }
            return head;
        }
        if (head.val == val) {
            head = head.next;
            head = removeElements(head, val);
        } else {
            head.next = removeElements(head.next, val);
        }
        return head;
    }

    class ListNode {
        int val;
        ListNode next;

        ListNode(int x) {
            val = x;
        }
    }

    @Test
    public void test() {
        ListNode listNode = new ListNode(1);
        listNode.next = new ListNode(2);
        listNode.next.next = new ListNode(3);
        listNode.next.next.next = new ListNode(3);
        System.out.println(removeElements(listNode, 3).toString());

    }
}
